package top.jojoace.GlobalExceptionAdvice_And_AutoURI.core.hack;

import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.servlet.mvc.method.RequestMappingInfo;
import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping;

import java.lang.reflect.Method;

/**
 * <h3>实现 控制类 根据 自身文件 所在位置 添加 URL 前缀。</h3>
 *
 * @author a1615
 */
public class AutoPrefixUrlMapping extends RequestMappingHandlerMapping {

    @Value("${controller.api-package}")
    private String apiPackagePath;

    /**
     *
     * @param method - 类的全路径类名 + .方法名()
     * @param handlerType - 类方法对应的响应路径
     * @return
     */
    @Override
    protected RequestMappingInfo getMappingForMethod(Method method, Class<?> handlerType) {
//        return super.getMappingForMethod(method, handlerType);

        RequestMappingInfo mappingInfo = super.getMappingForMethod(method,handlerType);

        if (mappingInfo != null){
            String prefix = this.getPrefix(handlerType);
            /**
             * RequestMappingInfo.paths(prefix).build() 创建一个新的RequestMappingInfo.Builder。
             * .build() 获得RequestMappingInfo。
             * 。combine(mappingInfo) 将获得的前缀路径与原对象方法上的路径进行拼接，获得新的RequestMappingInfo。
             */
            RequestMappingInfo newMappingInfo = RequestMappingInfo.paths(prefix).build().combine(mappingInfo);
            System.out.println(newMappingInfo);
            return newMappingInfo;
        }
        return  mappingInfo;
    }

    /**
     * 获得修改后的前缀路径
     * @param handlerType
     * @return 修改后的前缀路径
     *
     */
    private String getPrefix(Class<?> handlerType) {
        String packageName = handlerType.getPackage().getName();
        String dotPath = packageName.replaceAll(this.apiPackagePath,"");
        return dotPath.replace(".","/");
    }
}
